package com.lucy.viewer.vditor.service;

import cn.hutool.core.io.file.FileNameUtil;
import com.lucy.viewer.constant.ViewerConst;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;

/**
 * @author kindear
 * markdown 编辑器预处理服务
 */
@Slf4j
@Service
public class VditorPreHandlerService {
    public String handle(String url,String fileName){
        // 获取文件类型
        String fileType = FileNameUtil.extName(fileName);
        // 如果是图片
        if (ViewerConst.IMG_FILE.contains(fileType)){
            return "!["+fileName+"]("+url+")";
        }
        // 如果是音乐
        if (ViewerConst.MUSIC_FILE.contains(fileType)){
            String t =  "<audio id=\"audio\" controls=\"\" preload=\"none\">\n" +
                        "      <source id=\"mp3\" src=\"${url}\">\n" +
                        "</audio>";
            return t.replace("${url}",url);
        }

        // 其他直接超链接
        return "["+fileName+"]("+url+")";
    }
}
